package com.blog.aspect;

import com.blog.annotation.AutoFill;
import com.blog.enumeration.OperationType;
import lombok.extern.slf4j.Slf4j;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.Signature;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.stereotype.Component;

import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.time.LocalDateTime;

/**
 *  自定义切面类，通过前置通知来实现公共字段中共填充
 */

@Aspect
@Component // 加了这个注解，就会被加入到ioc容器中
@Slf4j
public class AutoFillAspect {

    /**
     * 切入点
     */
    @Pointcut("@annotation(com.blog.annotation.AutoFill)")
    public void autoFillPointCut(){}

    @Before("autoFillPointCut()")
    public void autoFill(JoinPoint joinPoint){
        // 先获取 方法签名 对象，才可以再获取方法上的数据
        MethodSignature signature = (MethodSignature) joinPoint.getSignature();

        // 获取方法上的注解，用于业务分支
        AutoFill autoFill = signature.getMethod().getAnnotation(AutoFill.class);
        OperationType operationType = autoFill.value(); // 获取方法的注解的值

        // 获取到拦截方法的需要填充信息的对象（默认在第一个参数）
        Object[] args = joinPoint.getArgs();
        if ( args.length == 0 || args == null) { // 如果参数中没有东西，直接返回
            return ;
        }

        Object entity = args[0]; // 获取参数数组中的实体对象

        // 先设置需要设置的值，当前时间，修改时间
        LocalDateTime now = LocalDateTime.now();

        // 业务分支，进行不同的字段注入
        if (operationType == OperationType.INSERT) {//当前进行 INSERT 业务的处理
            try {
    // 先获取实体对象中的 字段的set方法(getDeclaredMethod()可以获取当前类的所有方法； getMethod只能获取父类和当前类的public方法)
                // 第一个参数是方法名称，第二个参数是该方法的参数类型
                Method setCreateTime = entity.getClass().getDeclaredMethod("setCreateTime", LocalDateTime.class);
                Method setUpdateTime = entity.getClass().getDeclaredMethod("setUpdateTime", LocalDateTime.class);

                // 通过反射调用目标方法，第一个参数是可以调用目标方法的实体类，后续参数是目标方法的参数
                setCreateTime.invoke(entity, now);
                setUpdateTime.invoke(entity, now);
            } catch (Exception e) {
                e.printStackTrace();
            }
        } else if (operationType == OperationType.UPDATE) {
            try {
                Method setUpdateTime = entity.getClass().getDeclaredMethod("setUpdateTime", LocalDateTime.class);

                setUpdateTime.invoke(entity, now);
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
        log.info("业务开发完成");
    }
}
